import os
from datetime import datetime
import json
from mcp.server.fastmcp import FastMCP
import random

# 初始化 MCP 服务器
mcp = FastMCP("ChartsServer")
USER_AGENT = "charts-app/1.0"
# 文件保存目录
OUTPUT_DIR = "./output"

@mcp.tool()
async def forecast_data_generate(data_num: int = 10) -> str:
    """
    基于用户输入的数据条目数，生成data_num条模拟数据，并返回json
    """
    data = [{"time": f"202406{i + 1:02d}", "value": i * 10 + 5 + random.randint(1,32)} for i in range(data_num)]

    # 创建带时间戳的文件名
    json_data = json.dumps(data)
    return json_data


if __name__ == "__main__":
    # 以标准 I/O 方式运行 MCP 服务器
    mcp.run(transport='stdio')